package com.future.offer;

/**
 * Description: 剑指 Offer II 012. 左右两边子数组的和相等
 * <p>
 * 给你一个整数数组 nums ，请计算数组的 中心下标 。
 * 数组 中心下标 是数组的一个下标，其左侧所有元素相加的和等于右侧所有元素相加的和。
 * <p>
 * 如果中心下标位于数组最左端，那么左侧数之和视为 0 ，因为在下标的左侧不存在元素。这一点对于中心下标位于数组最右端同样适用。
 * <p>
 * 如果数组有多个中心下标，应该返回 最靠近左边 的那一个。如果数组不存在中心下标，返回 -1 。
 * <p>
 * 输入：nums = [2, 1, -1]
 * 输出：0
 * 解释：
 * 中心下标是 0 。
 * 左侧数之和 sum = 0 ，（下标 0 左侧不存在元素），
 * 右侧数之和 sum = nums[1] + nums[2] = 1 + -1 = 0 。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/tvdfij
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 * @author weiruibai.vendor
 * Date: 2023/1/20 10:27
 */
public class Solution_012 {

    private static Solution_012 instance = new Solution_012();

    public static void main(String[] args) {
        int[] nums = new int[]{1, 7, 3, 6, 5, 6};
        nums = new int[]{1, 2, 3};
        //nums = new int[]{2, 1, -1}; // 答案：0
        nums = new int[]{1, 7, 3, 6, 5, 6};
        nums = new int[]{-1, -1, 0, 1, 1, 0};

        System.out.println(instance.pivotIndex(nums));
    }

    public int pivotIndex(int[] nums) {
        int N = nums.length;
        if (N == 1) {
            return -1;
        }
        int ans = Integer.MAX_VALUE;
        /**
         * 左往右前缀和
         * preL2RSumArr[i]，计算：0,1,...,i-1的和
         */
        int[] preL2RSumArr = new int[N];
        preL2RSumArr[0] = 0;
        for (int i = 1; i < N; i++) {
            preL2RSumArr[i] = preL2RSumArr[i - 1] + nums[i - 1];
        }
        /**
         * 右往左前缀和
         * preR2LSumArr[i]，计算：N-1,N-2,...,i+1的和
         */
        int[] preR2LSumArr = new int[N];//
        preR2LSumArr[N - 1] = 0;
        for (int i = N - 2; i >= 0; i--) {
            preR2LSumArr[i] = preR2LSumArr[i + 1] + nums[i + 1];
        }
        /**
         * 那么位置i
         * preL2RSumArr[i]，表示：0,1,...,i-1的和
         * preR2LSumArr[i]，表示：N-1,N-2,...,i+1的和
         * 当preL2RSumArr[i]=preR2LSumArr[i]时，就是答案
         */
        for (int i = 0; i < N; i++) {
            if (preL2RSumArr[i] == preR2LSumArr[i]) {
                ans = Math.min(ans, i);
                break;
            }
        }
        return ans == Integer.MAX_VALUE ? -1 : ans;
    }

}
